import argparse

'''
training settings

metavar参数，用来控制部分命令行参数的显示
'''
parser = argparse.ArgumentParser(description='PyTorch Example for all')
parser.add_argument('--train-batch-size', type=int, default=30, metavar='N',
                    help='input batch size for training (default: 32)')
parser.add_argument('--test-batch-size', type=int, default=30, metavar='N',
                    help='input batch size for testing (default: 64)')
parser.add_argument('--epoches', type=int, default=500, metavar='N',
                    help='number of epochs to train (default: 10)')
parser.add_argument('--lr', type=float, default=0.001, metavar='LR',
                    help='learning rate (default: 0.0001)')
parser.add_argument('--momentum', type=float, default=0.5, metavar='M',
                    help='SGD momentum (default: 0.5)')
parser.add_argument('--seed', type=int, default=123, metavar='S',
                    help='random seed 设置种子的用意是一旦固定种子，后面依次生成的随机数其实都是固定的,有利于实验结果的产生与比较')
parser.add_argument('--use_cuda', type=bool, default=True,
                    help='whether to use cuda to accerlate')
parser.add_argument('--base_data_path', type=str, default='G:/数据集/分类/',
                    help="total base data path for training")
parser.add_argument('--resume', type=bool, default=True, metavar='R',
                    help="whether to use the pretrained model to start the train")
parser.add_argument('--saved_model', type=str, default="G:/项目/视频分类/3dcnn/weights_flow/",
                    help="the path to store the weight")
parser.add_argument('--val_num', type=float, default=0.3,
                    help="percentage of validate data")
parser.add_argument('--pretrained_weight', type=str, default="G:/项目/视频分类/3dcnn/weights_flow/Epoch-326-best_acc-0.7300000190734863_loss_0.69.pth",
                    help="the path to load the pytorch weight")
parser.add_argument('--save', type=bool, default=True,
                    help="whether to save the model weight")
parser.add_argument('--project_name', type=str, default='视频分类',
                    help="该项目的工程名称")
parser.add_argument('--use_aug', type=bool, default=False,
                    help='使用数据增广,增加数据多样性')
parser.add_argument('--model_name',type=str,default="shuffle_net",
                    help='model name')
parser.add_argument('--train_dir',type=str,default="G:/datasets/flow/train")
parser.add_argument('--test_dir',type=str,default="G:/datasets/flow/test")

parser.add_argument('--val_dir',type=str,default="G:/datasets/flow/val")